# !/usr/bin/env python
# -*-coding:utf-8 -*-

"""
# File       : my_pagiantion.py
# Time       ：2022/11/4 9:49
# Author     ：hhs
# version    ：python 3.6
# Description：
"""
from rest_framework.pagination import PageNumberPagination

from common.result import success


class MyPageNumber(PageNumberPagination):
    page_query_param = "page"
    page_size_query_param = 'pagesize'
    max_page_size = 10

    # 重写分页返回方法，按照指定的字段进行分页数据返回
    def get_paginated_response(self, data):
        return success({
            'count': self.page.paginator.count,  # 总数量
            'lists': data,  # 用户数据
            'page': self.page.number,  # 当前页数
            'pages': self.page.paginator.num_pages,  # 总页数
            'pagesize': self.page.paginator.per_page  # 后端指定的页容量
        })
